#存储班级和对应学生列表
class_student = {}
#添加班级
def add_class():
    class_name = input("请输入班级名称").strip()
    if class_name in class_student:
        print("班级已存在")
    else:
        class_student[class_name] = []
        print(f"班级'{class_name}'添加成功")
#班级列表
def class_li():
    if not class_student:
        print("暂无班级数据")
    else:
        for class_name in class_student:
            print(f"班级名称：{class_name}")
#添加学生，要有学生的基础信息，还要把学生添加到对应班级中
#一行注释
def add_student():
    if not class_student:
        print("暂无班级，请先添加班级")
        return
    class_li()
    class_name = input("请输入要加入的班级：").strip()
    if class_name not in class_student:
        print("班级不存在，请重新选择")
        return
    student_name = input("请输入学生姓名：").strip()
    student_id = input("请输入学生学号：").strip()
    student_phone = input("请输入学生联系方式：").strip()

    student_message = {
        "姓名":student_name,
        "学号":student_id,
        "联系方式":student_phone
    }
    class_student[class_name].append(student_message)
    print("添加成功")







while 1:
    print("========欢迎来到信思智学班级管理系统========")
    print("[1]班级添加，[2]班级列表，[3]学生添加，[4]学生列表，[5]安全退出")
    num = (input("请输入对应菜单编号---:"))
    if num == "1":
        print("===班级添加===")
        add_class()
    elif num == "2":
        print("===班级列表===")
        class_li()
    elif num == "3":
        print("===学生添加===")
        add_student()
    elif num == "4":
        print("===学生列表列表===")
    else:
        print("您已退出系统")
        break






